
import os
import sys
from PIL import Image

def create_image_directory():
    """创建image目录（如果不存在）"""
    image_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "image")
    if not os.path.exists(image_dir):
        os.makedirs(image_dir)
    return image_dir

def resize_images():
    """处理当前目录下所有PNG图片，生成64x64px的缩略图"""
    # 获取当前目录
    current_dir = os.path.dirname(os.path.abspath(__file__))

    # 创建image目录
    image_dir = create_image_directory()

    # 遍历当前目录下的所有文件
    for filename in os.listdir(current_dir):
        # 检查文件是否为PNG图片
        if filename.lower().endswith('.png') and not filename.startswith('thumbnail_'):
            try:
                # 打开图片
                img_path = os.path.join(current_dir, filename)
                img = Image.open(img_path)

                # 创建64x64的缩略图
                img.thumbnail((64, 64))

                # 生成新文件名（添加thumbnail_前缀）
                new_filename = f"thumbnail_{filename}"
                new_img_path = os.path.join(image_dir, new_filename)

                # 保存缩略图
                img.save(new_img_path, "PNG")
                print(f"已创建缩略图: {new_img_path}")

            except Exception as e:
                print(f"处理文件 {filename} 时出错: {str(e)}")

if __name__ == "__main__":
    resize_images()
    print("所有图片处理完成！")
